import pandas as pd
import requests
import json
# ===========================================================================================
listJZ = []  # 保存某基金的所有的历史净值
yes_jz = []  # 保存某基金最近更新的历史净值
yes_time = []  # 保存某基金最近更新的历史净值的时间


def get_gz_history(fund_code):
    url = f"http://api.fund.eastmoney.com/f10/lsjz?fundCode={fund_code}&pageIndex=1&pageSize=20"
    headers = {
        'Referer': 'http://fundf10.eastmoney.com',
        'User-Agent': 'Mozilla/5.0'
    }
    r = requests.get(url, headers=headers, timeout=20)
    data = r.json()['Data']['LSJZList']
    for get_ls_num in range(len(data)):
        listJZ.append(float(data[get_ls_num]['DWJZ']))  # 保存所有历史净值
    yes_jz.append(float(data[0]['DWJZ']))  # 保存最近更新的历史净值
    yes_time.append(data[0]['FSRQ'])  # 保存最近更新的历史净值的时间


if __name__ == '__main__':
    mon = input('买入的金额:')
    get_gz_history('023948')
    print(yes_jz, yes_time)
    df = pd.read_excel('../Fund_form_files/Y策略验证基金.xlsx')
    # 追加新数据
    new_data = {
        '买入时间': yes_time[0],
        '买入金额': int(mon),
        '买入净值': yes_jz[0],
        '买入份额': f'{int(mon) / yes_jz[0]:.2f}'
    }
    df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
    df.to_excel('../Fund_form_files/Y策略验证基金.xlsx', index=False)
    print(df)
